// import { audit } from './audit/audit'
import { beck } from './beck/beck'
import { blcs } from './blcs/blcs'
import { bwcs } from './bwcs/bwcs'
import { bmi_imperial, bmi_metric } from './bmi'
// import { breast_q as breast_q_conserving_therapy_pre_and_postoperative } from './breast_q/breast_conserving/breast_q'
// import { cade_q_sv } from './cade_q'
// import { caregiver_strain_index } from './caregiver_strain_index/caregiver_strain_index'
// import { cat } from './cat/cat'
// import { ccq } from './ccq/ccq'
// import { cdlqi } from './cdlqi/cdlqi'
// import { CHA2DS2_VASc_Score } from './CHA2DS2_VASc_Score/CHA2DS2_VASc_Score'
// import { chc_preop_brochure_triage } from './chc'
// import { femmes_enceintes_triage } from './chc/femmes_enceintes/triage/triage'
// import { comi_back, comi_neck } from './comi'
// import { compass_31 } from './compass_31/compass_31'
// import { constant_murley_score_orthotoolkit } from './constant_murley_score'
// import { core_om } from './core_om/core_om'
// import { cpdi } from './cpdi/cpdi'
// import { csi } from './csi/csi'
// import { dast_10 } from './dast_10/dast_10'
// import { demo_calculation } from './demo_calculation/demo_calculation'
// import { difference_calc } from './difference_calc/difference_calc'
// import { dlqi } from './dlqi/dlqi'
// import { dn4 } from './dn4/dn4'
// import { dri } from './dri/dri'
// import {
//   eortc_qlq_br23,
//   eortc_qlq_br45,
//   eortc_qlq_c30,
//   eortc_qlq_cr29,
//   eortc_qlq_lc13,
//   eortc_qlq_lc29,
//   eortc_qlq_pr25,
// } from './eortc'
// import { epic_26 } from './epic_26/epic_26'
// import { eq5d_3l, eq5d_5l } from './eq5d'
// import { ess } from './ess/ess'
// import { faam } from './faam/faam'
// import { short_fes_i } from './fes_i'
// import { fnd } from './fnd/fnd'
// import { foot_function_index_5pt } from './foot_function_index'
// import {
//   forgotten_joint_score_hip,
//   forgotten_joint_score_knee,
// } from './forgotten_joint_score_12'
// import { pss_4 } from './pss_4/pss_4'
// import { stop_bang } from './stop_bang/stop_bang'
// import { gad_2 } from './GAD_2/gad_2'
// import { gad_7 } from './GAD_7/gad_7'
// import { ghq_12 } from './ghq_12/ghq_12'
// import { hads } from './hads/HADS_score'
// import { harris_hip_score } from './harris_hip_score/harris_hip_score'
// import { hoos_extended } from './hoos_extended/hoos_extended'
// import { hoos_ps } from './hoos_ps/hoos_ps'
// import { hos } from './hos/hos'
// import { hrqol_4 } from './hrqol'
// import { ias } from './ias/ias'
// import { IBD_control } from './IBD_control/IBD_control'
// import { ibd_disk_total_score } from './ibd_disk_total_score/ibd_disk_total_score'
// import { iief5 } from './iief5/iief5'
// import { ikdc } from './ikdc/ikdc'
// import { ipss } from './ipss/ipss'
// import { isi } from './isi/isi'
// import { k_bild } from './k_bild/k_bild'
// import { KCCQ_12 } from './KCCQ_12/KCCQ_12'
// import { koos_ps } from './koos_ps/koos_ps'
// import { math_divide, math_multiply, math_subtract, math_sum } from './math'
// import { mds_updrs } from './mds_updrs/mds_updrs'
// import { mini_best_test } from './mini_best_test/mini_best_test'
// import { mfis } from './mfis/mfis'
// import { mmse } from './mmse/mmse'
// import { moca } from './moca/moca'
// import { modified_caregiver_strain_index } from './modified_caregiver_strain_index/modified_caregiver_strain_index'
// import { mpi } from './mpi/mpi'
// import { msq } from './msq/msq'
// import { ndi } from './ndi/ndi'
// import { oas } from './oas/oas'
// import { ompq, ompq_10 } from './orebro'
// import { oswestry } from './oswestry/oswestry'
// import { oxford_hip_score } from './oxford_hip_score/oxford_hip_score'
// import { oxford_knee_score } from './oxford_knee_score/oxford_knee_score'
// import { packyears } from './packyears/packyears'
// import { panss_6 } from './panss_6/panss_6'
// import { paq_c } from './paq_c/paq_c'
// import { pci } from './pci/pci'
// import { pcl_5 } from './pcl_5/pcl_5'
// import { pcs } from './pcs/pcs'
// import { pdi } from './pdi/pdi'
// import { pdq_8 } from './pdq_8/pdq_8'
// import { phq_2 } from './phq_2/phq_2'
// import { phq_4 } from './phq_4/phq_4'
// import { phq_8 } from './phq_8/phq_8'
// import { phq_9 } from './phq_9/phq_9'
// import { physical_activity_measurement } from './physical_activity_measurements/physical_activity_measurements'
// import { posas_observer } from './posas_observer/posas_observer'
// import { posas_patient } from './posas_patient/posas_patient'
// import { pro_ctcae } from './pro_ctcae/pro_ctcae'
// import { PRO2 } from './pro2/pro2'
// import { promis_10 } from './promis_10/promis_10'
// import { prtee } from './prtee/prtee'
// import { psk } from './psk/psk'
// import { qol_stoma } from './qol_stoma/qol_stoma'
// import { quickdash } from './quickdash/quickdash'
// import { sccai } from './sccai/sccai'
// import { scl90 } from './scl90/scl90'
// import { scl90r } from './scl90r/scl90r'
// import { sdq } from './sdq/sdq'
// import { sf12, sf36 } from './short form surveys'
// import { spadi } from './spadi/spadi'
// import { simple_shoulder_test } from './sst/simple_shoulder_test'
// import { start_back_screening_tool } from './start_back_screening_tool/start_back_screening_tool'
// import { tampa } from './tampa/tampa'
// import { visa_a, visa_g, visa_p } from './visa'
// import { yp_core } from './yp_core/yp_core'
// import { zarit_12 } from './zarit_12/zarit_12'
import { acro } from './acro/acro'
// import { korq } from './korq/korq'
// import { mlks } from './mlks/mlks'
// import { psqi } from './psqi/psqi'

export const CALCULATIONS: CalculationsLibraryType = {
  age_calc,
  acro,
  asrs,
  // audit,
  beck,
  bmi: bmi_metric,
  bmi_imperial,
  blcs,
  bwcs,
  // breast_q_conserving_therapy_pre_and_postoperative,
  // cade_q_sv,
  // caregiver_strain_index,
  // cat,
  // ccq,
  // cdlqi,
  // CHA2DS2_VASc_Score,
  // chc_preop_brochure_triage,
  // femmes_enceintes_triage,
  // comi_back,
  // comi_neck,
  // compass_31,
  // constant_murley_score_orthotoolkit,
  // core_om,
  // cpdi,
  // csi,
  // dast_10,
  // demo_calculation,
  // difference_calc,
  // dlqi,
  // dn4,
  // dri,
  // eortc_qlq_br23,
  // eortc_qlq_br45,
  // eortc_qlq_c30,
  // eortc_qlq_cr29,
  // eortc_qlq_lc13,
  // eortc_qlq_lc29,
  // eortc_qlq_pr25,
  // epic_26,
  // eq5d_3l,
  // eq5d_5l,
  // ess,
  // faam,
  // fnd,
  // foot_function_index_5pt,
  // forgotten_joint_score_hip,
  // forgotten_joint_score_knee,
  // gad_2,
  // gad_7,
  // ghq_12,
  // hads,
  // harris_hip_score,
  // hoos_extended,
  // hoos_ps,
  // hos,
  // hrqol_4,
  // ias,
  // IBD_control,
  // ibd_disk_total_score,
  // iief5,
  // ikdc,
  // isi,
  // ipss,
  // k_bild,
  // KCCQ_12,
  // koos_ps,
  // korq,
  // math_divide,
  // math_multiply,
  // math_sum,
  // math_subtract,
  // mds_updrs,
  // mini_best_test,
  // mfis,
  // mlks,
  // mmse,
  // moca,
  // modified_caregiver_strain_index,
  // mpi,
  // msq,
  // ndi,
  // oas,
  // ompq,
  // ompq_10,
  // oswestry,
  // oxford_hip_score,
  // oxford_knee_score,
  // packyears,
  // panss_6,
  // paq_c,
  // pci,
  // pcl_5,
  // pcs,
  // pdi,
  // pdq_8,
  // phq_2,
  // phq_4,
  // phq_8,
  // phq_9,
  // physical_activity_measurement,
  // posas_observer,
  // posas_patient,
  // pro_ctcae,
  // PRO2,
  // promis_10,
  // prtee,
  // psk,
  // psqi,
  // pss_4,
  // qol_stoma,
  // quickdash,
  // sccai,
  // scl90,
  // scl90r,
  // sdq,
  // sf12,
  // sf36,
  // short_fes_i,
  // spadi,
  // simple_shoulder_test,
  // start_back_screening_tool,
  // stop_bang,
  // tampa,
  ten_meter_walk_test,
  // visa_a,
  // visa_g,
  // visa_p,
  // yp_core,
  // zarit_12,

import { ASRS_PARTS, ASRS_SUBSCALES } from './definition'
import { Calculation } from '../../../api/shared/classes/Calculation'
import { ZodError } from 'zod'

const asrs_calculation = new Calculation(asrs)

describe('asrs', function () {
  it('asrs calculation function should be available as a calculation', function () {
    expect(CALCULATIONS).to.have.property('asrs')
  })

  describe('basic assumptions', function () {
    const outcome = asrs_calculation.calculate({ payload: best_response })

    it('should have the expected calculation ids', function () {
      const EXPECTED_CALCULATION_IDS = [
        'ASRS_PART_A_SCORE',
        'ASRS_PART_B_SCORE',
        'ASRS_TOTAL_SCORE',
        'ASRS_INATTENTIVE_SUBSCALE_SCORE',
        'ASRS_HYPERACTIVE_IMPULSIVE_SUBSCALE_MOTOR_SCORE',
        'ASRS_HYPERACTIVE_IMPULSIVE_SUBSCALE_VERBAL_SCORE',
      ]
      const configured_calculation_ids = Object.keys(outcome)

      expect(configured_calculation_ids).to.eql(EXPECTED_CALCULATION_IDS)
    })
  })

  describe('validation', function () {
    describe('the score includes the correct input fields', function () {
      it('should have all the expected input ids configured', function () {
        const EXPECTED_INPUT_IDS = [
          'Q01',
          'Q02',
          'Q03',
          'Q04',
          'Q05',
          'Q06',
          'Q07',
          'Q08',
          'Q09',
          'Q10',
          'Q11',
          'Q12',
          'Q13',
          'Q14',
          'Q15',
          'Q16',
          'Q17',
          'Q18',
        ]

        const configured_input_ids = Object.keys(
          asrs_calculation.inputSchema.shape
        )

        expect(EXPECTED_INPUT_IDS).to.eql(configured_input_ids)
      })
    })

    describe('ASRS parts', function () {
      describe('Part A', function () {
        it('should have the expected input ids', function () {
          const EXPECTED_INPUT_IDS = ['Q01', 'Q02', 'Q03', 'Q04', 'Q05', 'Q06']

          expect(EXPECTED_INPUT_IDS).to.eql(ASRS_PARTS.PART_A)
        })
      })

      describe('Part B', function () {
        it('should have the expected input ids', function () {
          const EXPECTED_INPUT_IDS = [
            'Q07',
            'Q08',
            'Q09',
            'Q10',
            'Q11',
            'Q12',
            'Q13',
            'Q14',
            'Q15',
            'Q16',
            'Q17',
            'Q18',
          ]

          expect(EXPECTED_INPUT_IDS).to.eql(ASRS_PARTS.PART_B)
        })
      })
    })

    describe('ASRS subscales', function () {
      describe('Inattentive subscale', function () {
        it('should have the expected input ids', function () {
          const EXPECTED_INPUT_IDS = [
            'Q01',
            'Q02',
            'Q03',
            'Q04',
            'Q07',
            'Q08',
            'Q09',
            'Q10',
            'Q11',
          ]

          expect(EXPECTED_INPUT_IDS).to.eql(ASRS_SUBSCALES.INATTENTIVE_SUBSCALE)
        })
      })

      describe('Hyperactive/Impulsive subscale (Motor)', function () {
        it('should have the expected input ids', function () {
          const EXPECTED_INPUT_IDS = ['Q05', 'Q06', 'Q12', 'Q13', 'Q14']

          expect(EXPECTED_INPUT_IDS).to.eql(
            ASRS_SUBSCALES.HYPERACTIVE_IMPULSIVE_SUBSCALE_MOTOR
          )
        })
      })

      describe('Hyperactive/Impulsive subscale (Verbal)', function () {
        it('should have the expected input ids', function () {
          const EXPECTED_INPUT_IDS = ['Q15', 'Q16', 'Q17', 'Q18']

          expect(EXPECTED_INPUT_IDS).to.eql(
            ASRS_SUBSCALES.HYPERACTIVE_IMPULSIVE_SUBSCALE_VERBAL
          )
        })
      })
    })

    describe('when an answer is below the expected range', function () {
      it('should throw an InvalidInputsError', function () {
        expect(() =>
          asrs_calculation.calculate({
            payload: {
              Q01: -1,
            },
          })
        ).to.throw(ZodError)
      })
    })

    describe('when an answer is above the expected range', function () {
      it('should throw an InvalidInputsError', function () {
        expect(() =>
          asrs_calculation.calculate({
            payload: {
              Q01: 5,
            },
          })
        ).to.throw(ZodError)
      })
    })

    describe('when there are non-numerical answers', function () {
      it('should throw an InvalidInputsError', function () {
        expect(() =>
          asrs_calculation.calculate({
            payload: {
              Q01: "I'm not a number",
            },
          })
        ).to.throw(ZodError)
      })

import { Calculation } from '../../../api/shared/classes/Calculation'
import { ZodError } from 'zod'

const tmwt_calculation = new Calculation(ten_meter_walk_test)

describe('ten_meter_walk_test', function () {
  it('ten_meter_walk_test calculation function should be available as a calculation', function () {
    expect(CALCULATIONS).to.have.property('ten_meter_walk_test')
  })

  describe('specific_steps_ten_meter_walk_test_calc', function () {
    describe('basic assumptions', function () {
      const outcome = tmwt_calculation.calculate({
        payload: {
          TRIAL_1: 10,
        },
      })

      it('should return 3 calculation results', function () {
        const AMOUNT_OF_RESULTS = 3
        expect(Object.keys(outcome)).to.have.length(AMOUNT_OF_RESULTS)
      })

      it('should the calculation names', function () {
        const EXPECTED_CALCULATION_IDS = [
          'TMWT_MEAN_IN_SECONDS',
          'TMWT_MEAN_IN_METERS_PER_SECOND',
          'TMWT_MEAN_IN_KILOMETERS_PER_HOUR',
        ]

        const EXTRACTED_CALCULATION_IDS_FROM_RESULT = Object.keys(outcome)

        expect(EXTRACTED_CALCULATION_IDS_FROM_RESULT).to.eql(
          EXPECTED_CALCULATION_IDS
        )
      })
    })

    describe('validations', function () {
      describe('the score includes the correct input fields', function () {
        it('should have all the expected input ids configured', function () {
          const EXPECTED_INPUT_IDS = ['TRIAL_1', 'TRIAL_2', 'TRIAL_3']
          const configured_calculation_input_ids = Object.keys(
            tmwt_calculation.inputSchema.shape
          )

          expect(EXPECTED_INPUT_IDS).to.eql(configured_calculation_input_ids)
        })
      })

      describe('when called with a response with answers out of the expected [0,4] range', function () {
        describe('when an answer is not a number', function () {
          it('should throw an error', function () {
            expect(() =>
              tmwt_calculation.calculate({
                payload: {
                  TRIAL_1: 'hello',
                },
              })
            ).to.throw(ZodError)
          })
        })

        describe('when the response is empty', function () {
          it('should throw an error', function () {
            expect(() =>
              tmwt_calculation.calculate({
                payload: {},
              })
            ).to.throw(ZodError)
          })
        })
      })
    })

    describe('score calculation', function () {
      describe('when called with a single trial', function () {
        const outcome = tmwt_calculation.calculate({
          payload: {
            TRIAL_1: 7, // Took the patient 7 seconds to walk 10 meters
          },
        })

        it('should return the correct result for the mean in seconds', function () {
          const EXPECTED_RESULT = 7
          expect(outcome.TMWT_MEAN_IN_SECONDS).to.eql(EXPECTED_RESULT)
        })

        it('should return the correct result for the mean in meters per second', function () {
          const EXPECTED_RESULT = 1.43
          expect(outcome.TMWT_MEAN_IN_METERS_PER_SECOND).to.eql(EXPECTED_RESULT)
        })

        it('should return the correct result for the mean in kilometers per hour', function () {
          const EXPECTED_RESULT = 5.14
          expect(outcome.TMWT_MEAN_IN_KILOMETERS_PER_HOUR).to.eql(
            EXPECTED_RESULT
          )
        })
      })